from __future__ import absolute_import
from __future__ import print_function

import os
import json
from keras.optimizers import Adam, SGD
from meta import TrainingMetadata
from experiment import TRAINING_JSON, META_DIR

def get_training_metadata():
    training_metadata = TrainingMetadata()
    training_metadata.optimizer = SGD(momentum=0.9, decay=0.9, lr=0.01)
    training_metadata.loss = 'mse'
    training_metadata.batch_size = 48
    training_metadata.nb_epoch = 5
    training_metadata.nb_iter = 1
    training_metadata.shuffle = False
    training_metadata.mode = 'classify'
    training_metadata.class_weights = {0:13.458,1:1.4}

    return training_metadata
